#!/bin/bash

# model type = {
#     naive,
#     centre,
#     a-softmax,
#     am-softmax,
#     normface,
#     c-contrastive,
#     c-triplet
# }

echo naive
for bias in 0 1; do
    # clear
    echo === model type: naive, bias: $bias ===
    python main.py --model_type naive --bias $bias --Lambda 0
done
python test.naive.py --model_type naive

echo centre
for lamb in 0.0001 0.01 0.1 1; do
    echo === model type: centre, lambda: $lamb ===
    python main.py --model_type centre --bias 0 --Lambda $lamb
done
python test.centre.py --model_type centre

echo a-softmax
for m in 1 2 3 4; do
    python main.py --model_type a-softmax --bias 0 --Lambda 0 --margin $m \
        --lr 0.01 --momentum 0.5 --batch_size 32 --weight_decay 0.00001
done
python test.a-softmax.py --model_type a-softmax

echo normface
python main.py --model_type normface --bias 0 --Lambda 0 --scale 20
python test.normface.py --model_type normface

echo am-softmax
python main.py --model_type am-softmax --bias 0 --Lambda 0 --scale 5 --train_scale 0 --margin 0.4
python test.am-softmax.py --model_type am-softmax

# echo c-contrastive
# python main.py --model_type c-contrastive --bias 0 --Lambda 0 --margin 1 --lr 0.001

# echo c-triplet
# python main.py --model_type c-triplet --bias 0 --Lambda 0

echo ablation study
python ablation.py

